library(tidyverse)

joined_raw_owd <- readRDS(here::here("output/joined_raw_owd.Rds"))

countries <- joined_raw_owd$country %>% unique() %>% sort()

vis_miss_cntry <- function(cntry) {
  p <- joined_raw_owd %>% 
    filter(country == cntry) %>% 
    select(-c(starts_with("suicide"), country, iso3c, banking_crisis:inflation_crisis, average_household_size_number_of_members:`6_or_more_members`, alc_wine, alc_beer, alc_other_alcoholic_beverages, alc_spirits)) %>% 
    mutate(across(-year, ~ is.na(.x))) %>% 
    pivot_longer(-year, values_to = "missing") %>% 
    ggplot(aes(year, name, fill = fct_rev(as.factor(missing)))) +
    geom_raster() +
    coord_cartesian(xlim = range(joined_raw_owd$year)) +
    labs(x = NULL, y = NULL, fill = "Missing", subtitle = cntry) +
    theme_minimal() +
    theme(panel.ontop = TRUE)
  
  print(p)
}

walk(countries, function(x) {
  cat("###", x, "\n\n")
  vis_miss_cntry(x)
  cat("\n\n")
})

Afghanistan

Albania

Algeria

American Samoa

Andorra

Angola

Antigua & Barbuda

Argentina

Armenia

Australia

Austria

Azerbaijan

Bahamas

Bahrain

Bangladesh

Barbados

Belarus

Belgium

Belize

Benin

Bermuda

Bhutan

Bolivia

Bosnia & Herzegovina

Botswana

Brazil

Brunei

Bulgaria

Burkina Faso

Burundi

Cambodia

Cameroon

Canada

Cape Verde

Central African Republic

Chad

Chile

China

Colombia

Comoros

Congo - Brazzaville

Congo - Kinshasa

Costa Rica

Côte d’Ivoire

Croatia

Cuba

Cyprus

Czechia

Denmark

Djibouti

Dominica

Dominican Republic

Ecuador

Egypt

El Salvador

Equatorial Guinea

Eritrea

Estonia

Eswatini

Ethiopia

Fiji

Finland

France

Gabon

Gambia

Georgia

Germany

Ghana

Greece

Greenland

Grenada

Guam

Guatemala

Guinea

Guinea-Bissau

Guyana

Haiti

Honduras

Hungary

Iceland

India

Indonesia

Iran

Iraq

Ireland

Israel

Italy

Jamaica

Japan

Jordan

Kazakhstan

Kenya

Kiribati

Kuwait

Kyrgyzstan

Laos

Latvia

Lebanon

Lesotho

Liberia

Libya

Lithuania

Luxembourg

Madagascar

Malawi

Malaysia

Maldives

Mali

Malta

Marshall Islands

Mauritania

Mauritius

Mexico

Micronesia (Federated States of)

Moldova

Mongolia

Montenegro

Morocco

Mozambique

Myanmar (Burma)

Namibia

Nepal

Netherlands

New Zealand

Nicaragua

Niger

Nigeria

North Korea

North Macedonia

Northern Mariana Islands

Norway

Oman

Pakistan

Palestinian Territories

Panama

Papua New Guinea

Paraguay

Peru

Philippines

Poland

Portugal

Puerto Rico

Qatar

Romania

Russia

Rwanda

Samoa

São Tomé & Príncipe

Saudi Arabia

Senegal

Serbia

Seychelles

Sierra Leone

Singapore

Slovakia

Slovenia

Solomon Islands

Somalia

South Africa

South Korea

South Sudan

Spain

Sri Lanka

St. Lucia

St. Vincent & Grenadines

Sudan

Suriname

Sweden

Switzerland

Syria

Taiwan

Tajikistan

Tanzania

Thailand

Timor-Leste

Togo

Tonga

Trinidad & Tobago

Tunisia

Turkey

Turkmenistan

U.S. Virgin Islands

Uganda

Ukraine

United Arab Emirates

United Kingdom

United States

Uruguay

Uzbekistan

Vanuatu

Venezuela

Vietnam

Yemen

Zambia

Zimbabwe